FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime


WORKDIR /root
COPY . .
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip install jupyter
RUN jupyter notebook --generate-config
RUN mv jupyter_notebook_config.py  /root/.jupyter/jupyter_notebook_config.py


EXPOSE 6006
# 设置时区为东八区
ENV TZ=Asia/Shanghai
WORKDIR /root/workspace
CMD ["jupyter notebook --ip=0.0.0.0 --allow-root --port=6006 --no-browser"]